package com.atguigu.common.exception;

import com.atguigu.common.result.R;
import com.atguigu.common.result.ResponseEnum;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.ConversionNotSupportedException;
import org.springframework.beans.TypeMismatchException;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.http.converter.HttpMessageNotWritableException;
import org.springframework.jdbc.BadSqlGrammarException;
import org.springframework.web.HttpMediaTypeNotAcceptableException;
import org.springframework.web.HttpMediaTypeNotSupportedException;
import org.springframework.web.HttpRequestMethodNotSupportedException;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.MissingPathVariableException;
import org.springframework.web.bind.MissingServletRequestParameterException;
import org.springframework.web.bind.ServletRequestBindingException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.context.request.async.AsyncRequestTimeoutException;
import org.springframework.web.multipart.support.MissingServletRequestPartException;
import org.springframework.web.servlet.NoHandlerFoundException;

/**
 * @author Alice
 * @since 2021-08-27 星期五 17:49
 */
@Slf4j
@RestControllerAdvice
public class UnifiedExceptionHandler {

  @ExceptionHandler({BusinessException.class})
  public R handleException(BusinessException e) {
    log.error(e.getMessage(), e);
    return R.error().data("result", false).message(e.getMessage()).code(e.getCode());
  }

  @ExceptionHandler({BadSqlGrammarException.class})
  public R handleException(BadSqlGrammarException e) {
    log.error(e.getMessage(), e);
    return R.result(ResponseEnum.BAD_SQL_GRAMMAR_ERROR);
  }

  @ExceptionHandler({Exception.class})
  public R handleException(Exception e) {
    log.error(e.getMessage(), e);
    return R.error();
  }

  @ExceptionHandler({
    NoHandlerFoundException.class,
    HttpRequestMethodNotSupportedException.class,
    HttpMediaTypeNotSupportedException.class,
    MissingPathVariableException.class,
    MissingServletRequestParameterException.class,
    MissingServletRequestPartException.class,
    TypeMismatchException.class,
    HttpMessageNotReadableException.class,
    HttpMessageNotWritableException.class,
    MethodArgumentNotValidException.class,
    HttpMediaTypeNotAcceptableException.class,
    ServletRequestBindingException.class,
    ConversionNotSupportedException.class,
    AsyncRequestTimeoutException.class
  })
  public R handleServletException(Exception e) {
    log.error(e.getMessage(), e);
    return R.result(ResponseEnum.SERVLET_ERROR);
  }

}
